FROM dockerhub.genostack.com:8090/small-tool/diffdock:2.0.0

WORKDIR /opt/nim/workspace/diffdock
RUN ln -s /opt/nim/.cache/ngc/hub/models--nim--mit--diffdock/blobs/a3e20771968799c39e5e644a6388bd17 config.pbtxt &&\
	mkdir 1

WORKDIR /opt/nim/workspace/diffdock/1
RUN ln -s /opt/nim/.cache/ngc/hub/models--nim--mit--diffdock/blobs/56703cb54ae554127280586cb929bd9f model.py &&\
	mkdir workdir

WORKDIR /opt/nim/workspace/diffdock/1/workdir
RUN mkdir diffdock_models_v1 diffdock_models_v2 esm_models

WORKDIR /opt/nim/workspace/diffdock/1/workdir/diffdock_models_v1
RUN ln -s /opt/nim/.cache/ngc/hub/models--nim--mit--diffdock/blobs/1bfc8d513817b5108d25927841c703cd confidence_parameters.yml
RUN ln -s /opt/nim/.cache/ngc/hub/models--nim--mit--diffdock/blobs/92917f751ff8d98865594ea9829c8a2e-4 confidence.pt
RUN ln -s /opt/nim/.cache/ngc/hub/models--nim--mit--diffdock/blobs/0af5e1df38b9282dc114cb6ed4cd752a score_parameters.yml
RUN ln -s /opt/nim/.cache/ngc/hub/models--nim--mit--diffdock/blobs/36c5eec79d3695a3f805767ea81fc34b-16 score.pt
WORKDIR /opt/nim/workspace/diffdock/1/workdir/diffdock_models_v2
RUN ln -s /opt/nim/.cache/ngc/hub/models--nim--mit--diffdock/blobs/76ccee7ff1bccdf38314900551111432 confidence_parameters.yml
RUN ln -s /opt/nim/.cache/ngc/hub/models--nim--mit--diffdock/blobs/a3ddffc00e099ceb4ef9c375ecb88333-4 confidence.pt
RUN ln -s /opt/nim/.cache/ngc/hub/models--nim--mit--diffdock/blobs/7ce810d86cd7b8886f796d9b7856dc02 score_parameters.yml
RUN ln -s /opt/nim/.cache/ngc/hub/models--nim--mit--diffdock/blobs/97d2e8f176f9a9c12f7ca3e6b87ef876-16 score.pt
WORKDIR /opt/nim/workspace/diffdock/1/workdir/esm_models
RUN ln -s /opt/nim/.cache/ngc/hub/models--nim--mit--diffdock/blobs/6194f061e75f37a60c436775f1f6be13 esm2_t33_650M_UR50D-contact-regression.pt
RUN ln -s /opt/nim/.cache/ngc/hub/models--nim--mit--diffdock/blobs/625aeb252d5f4447564f0edc75b2c8ec-497 esm2_t33_650M_UR50D.pt

WORKDIR /opt/nim

# 给 Tritonserver 更多的启动时间
COPY diffdock/opt/nim/inference.py /opt/nim/inference.py

# docker  build -t dockerhub.genostack.com:8090/small-tool/diffdock:2.0.0_linked .

